package main.leetcode.clockin.July;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;

/**
 * 350. 两个数组的交集 II
 *
 * <p>给定两个数组，编写一个函数来计算它们的交集。
 *
 * <p>示例 1: 输入: nums1 = [1,2,2,1], nums2 = [2,2] 输出: [2,2]
 *
 * <p>示例 2: 输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出: [4,9]
 *
 * <p>说明：输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。 我们可以不考虑输出结果的顺序。 进阶:
 *
 * <p>如果给定的数组已经排好序呢？你将如何优化你的算法？ 如果 nums1 的大小比 nums2 小很多，哪种方法更优？ 如果 nums2
 * 的元素存储在磁盘上，磁盘内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？
 */
public class day13 {
    public static void main(String[] args) {
        System.out.println(
                Arrays.toString(new day13().intersect(new int[] {1, 2, 2, 1}, new int[] {2, 2})));
    }

    /** HashMap */
    public int[] intersect1(int[] nums1, int[] nums2) {
        if (nums1 == null || nums2 == null || nums1.length == 0 || nums2.length == 0) {
            return new int[] {};
        }
        if (nums1.length > nums2.length) {
            int[] tmp = nums1;
            nums1 = nums2;
            nums2 = tmp;
        }
        HashMap<Integer, Integer> map1 = new HashMap<>();
        for (int i : nums1) {
            map1.put(i, map1.getOrDefault(i, 0) + 1);
        }
        List<Integer> res = new ArrayList<>();
        for (Integer i : nums2) {
            int cnt = map1.getOrDefault(i, 0);
            if (cnt > 0) {
                res.add(i);
                map1.put(i, --cnt);
            }
        }
        return res.stream().mapToInt(value -> value).toArray();
    }

    /** 排序 + 双指针 */
    public int[] intersect(int[] nums1, int[] nums2) {
        if (nums1 == null || nums2 == null || nums1.length == 0 || nums2.length == 0) {
            return new int[] {};
        }
        int m = nums1.length;
        int n = nums2.length;
        int i = 0, j = 0;
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        List<Integer> res = new ArrayList<>();
        while (i < m && j < n) {
            if (nums1[i] == nums2[j]) {
                res.add(nums1[i]);
                ++i;
                ++j;
            } else if (nums1[i] < nums2[j]) {
                ++i;
            } else {
                ++j;
            }
        }
        return res.stream().mapToInt(value -> value).toArray();
    }
}
